import { i18n } from '/@/locales/setupI18n';

type I18nGlobalTranslation = {
  (key: string): string;
  (key: string, locale: string): string;
  (key: string, locale: string, list: unknown[]): string;
  (key: string, locale: string, named: Record<string, unknown>): string;
  (key: string, list: unknown[]): string;
  (key: string, named: Record<string, unknown>): string;
};

type I18nTranslationRestParameters = [string, any];

function getKey(namespace: string | undefined, key: string) {
  if (!namespace) {
    return key;
  }
  if (key.startsWith(namespace)) {
    return key;
  }
  return `${namespace}.${key}`;
}

function formatText(key: string, args: any[]) {
  if (!key) return key;
  if (!args || args.length <= 0) return key;
  let text = key;
  if (arguments.length > 0) {
    for (const i in args) {
      if (args[i] != undefined) {
        text = text.replace('{' + i + '}', args[i]);
      }
    }
  }
  return text;
}

export function useI18n(namespace?: string): {
  t: I18nGlobalTranslation;
} {
  const normalFn = {
    t: (key: string) => {
      return getKey(namespace, key);
    },
  };

  if (!i18n) {
    return normalFn;
  }

  const { t, ...methods } = i18n.global;

  const tt = t as (arg0: string, ...arg: I18nTranslationRestParameters) => string;

  const tFn: I18nGlobalTranslation = (key: string, ...arg: any[]) => {
    if (!key) return '';
    const k = getKey(key.startsWith('common.') ? '' : namespace, key);
    const v = tt(k, ...(arg as I18nTranslationRestParameters));
    if (k == v) {
      const k2 = getKey('common', key);
      const v2 = tt(k2, ...(arg as I18nTranslationRestParameters));
      if (k2 == v2) return formatText(key, arg);
      return v2;
    }
    return v;
  };
  return {
    ...methods,
    t: tFn,
  };
}

// Why write this function？
// Mainly to configure the vscode i18nn ally plugin. This function is only used for routing and menus. Please use useI18n for other places

// 为什么要编写此函数？
// 主要用于配合vscode i18nn ally插件。此功能仅用于路由和菜单。请在其他地方使用useI18n
export const t = (key: string) => key;
